package com.sali.agent.impl;

import com.sali.agent.IPromptAgent;
import com.sali.agent.PromptAgentLoader;
import com.sali.annotation.CommonPromptDef;
import com.sali.constants.PromptAiModelNameConstants;
import com.sali.enums.PromptBusinessNameEnum;
import com.sali.enums.PromptLangNameEnum;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

/**
 * @description:
 * @author: sali
 * @date: 2024/9/1 下午3:33.
 * @version: 1.0
 */
@Component
public class LlamaChatPromptAgent implements IPromptAgent {

    @Autowired
    private PromptAgentLoader promptAgentLoader;

    @CommonPromptDef(aiModelName = PromptAiModelNameConstants.LLAMA)
    @Override
    public void init() {}

    @Override
    public String getPromptContent(PromptBusinessNameEnum promptBusinessNameEnum, PromptLangNameEnum promptLangNameEnum) {
        return promptAgentLoader.getLlmCommonPromptContent(this.getClass().getName(), promptBusinessNameEnum.getName(), promptLangNameEnum.getName());
    }

}
